class Solution {
public:
    int findPoisonedDuration(vector<int>& timeSeries, int duration) {
        int ret = 0;
        for (int i = 0; i < timeSeries.size() - 1; i++) {
            int x = timeSeries[i + 1] - timeSeries[i];
            if (x < duration) ret += x;
            else ret += duration;
        }
        return ret + duration;
    }
};